from db import DataFile
# 创建DataFile对象，csv文件名为test，
# 数据项为2个name和sex
d = DataFile('udata', ['name', 'sex'])
# 插入数据
d.insert(name='张三', sex='男')
d.insert(name='李四', sex='女')
d.insert(name='王五', sex='男')
# 保存数据
d.save()
# 查询数据
print(d.select())
print(d.select({'sex': '女'}))
# 更新数据，并查看结果
d.update({'name': '王五'}, sex='女')
print(d.select({'name': '王五'}))
# 删除数据，并查看结果
d.delete({'sex': '男'})
print(d.select())

from db import BaseEntity
t = BaseEntity('udata', ['name', 'sex'])
t.add_one(name='张三1', sex='男')
t.add_one(name='李四1', sex='女')
t.add_one(name='王五1', sex='男')
print(t.get({'name': '张三1'}))
print(t.get_all())

from model import FormattedMessage, IDExistException
# 创建FormattedMessage对象
f = FormattedMessage()
# 添加一条数据
try:
    # 添加时如果没有提供'id'，则自动生成一个唯一的id
    # 如果提供了重复的id，则抛出IDExistException异常        
    f.add_one(name='Test', text='This is a test message', 
              elementlist=['elem1', 'elem2'], category='info')  
except IDExistException as e:
    print(f"Error: {e}")
# 查询所有数据
print(f.get_all())